package cn.iocoder.yudao.module.dr.service.data;

import cn.iocoder.yudao.module.dr.controller.admin.data.vo.DataPageReqVO;
import cn.iocoder.yudao.module.dr.controller.admin.data.vo.DataSaveReqVO;
import cn.iocoder.yudao.module.dr.dal.dataobject.data.DataItemDO;
import cn.iocoder.yudao.module.dr.dal.mysql.data.DataItemMapper;
import org.springframework.stereotype.Service;
import jakarta.annotation.Resource;
import org.springframework.validation.annotation.Validated;
import org.springframework.transaction.annotation.Transactional;

import java.util.*;
import cn.iocoder.yudao.module.dr.controller.admin.data.vo.*;
import cn.iocoder.yudao.module.dr.dal.dataobject.data.DataDO;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;

import cn.iocoder.yudao.module.dr.dal.mysql.data.DataMapper;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.module.dr.enums.ErrorCodeConstants.*;

/**
 * 数据共享 Service 实现类
 *
 * @author 管理员
 */
@Service
@Validated
public class DataServiceImpl implements DataService {

    @Resource
    private DataMapper dataMapper;
    @Resource
    private DataItemMapper dataItemMapper;

    @Override
    @Transactional(rollbackFor = Exception.class)
    public Long createData(DataSaveReqVO createReqVO) {
        // 插入
        DataDO data = BeanUtils.toBean(createReqVO, DataDO.class);
        dataMapper.insert(data);

        // 插入子表
        createDataItemList(data.getId(), createReqVO.getDataItems());
        // 返回
        return data.getId();
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void updateData(DataSaveReqVO updateReqVO) {
        // 校验存在
        validateDataExists(updateReqVO.getId());
        // 更新
        DataDO updateObj = BeanUtils.toBean(updateReqVO, DataDO.class);
        dataMapper.updateById(updateObj);

        // 更新子表
        updateDataItemList(updateReqVO.getId(), updateReqVO.getDataItems());
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void deleteData(Long id) {
        // 校验存在
        validateDataExists(id);
        // 删除
        dataMapper.deleteById(id);

        // 删除子表
        deleteDataItemByDataId(id);
    }

    private void validateDataExists(Long id) {
        if (dataMapper.selectById(id) == null) {
            throw exception(DATA_NOT_EXISTS);
        }
    }

    @Override
    public DataDO getData(Long id) {
        return dataMapper.selectById(id);
    }

    @Override
    public PageResult<DataDO> getDataPage(DataPageReqVO pageReqVO) {
        return dataMapper.selectPage(pageReqVO);
    }

    // ==================== 子表（共享数据项） ====================

    @Override
    public List<DataItemDO> getDataItemListByDataId(Long dataId) {
        return dataItemMapper.selectListByDataId(dataId);
    }

    private void createDataItemList(Long dataId, List<DataItemDO> list) {
        list.forEach(o -> o.setDataId(dataId));
        dataItemMapper.insertBatch(list);
    }

    private void updateDataItemList(Long dataId, List<DataItemDO> list) {
        deleteDataItemByDataId(dataId);
		list.forEach(o -> o.setId(null).setUpdater(null).setUpdateTime(null)); // 解决更新情况下：1）id 冲突；2）updateTime 不更新
        createDataItemList(dataId, list);
    }

    private void deleteDataItemByDataId(Long dataId) {
        dataItemMapper.deleteByDataId(dataId);
    }

}